import numpy as np
import matplotlib.pyplot as plt

# 绘制多个坐标系
# index  第几个坐标系，从1开始


# 抛物线
x1 = np.linspace(-10,10,31)
# print(x1)
y1 = x1 ** 2
# 生成坐标系
# 创建坐标系之后，紧跟的用plt对象画图的操作就是当前坐标系
plt.subplot(1,3,1)
# 设置数据
plt.plot(x1,y1)

# 返回到变量中可方便调用，即给坐标系命名
# ax1 = plt.subplot(1,3,1)
# ax1.plot(x1,y1)

# 直线
x2 = np.linspace(1,10,15)
y2 = 2 * x2
plt.subplot(1,3,2)
plt.plot(x2,y2)

# 正弦曲线
x3 = np.linspace(0,10,100)
y3 = np.sin(x3)
plt.subplot(1,3,3)
plt.plot(x3,y3)

# 余弦曲线
# x4 = np.linspace(0,10,100)
# y4 = np.cos(x4)
# plt.subplot(2,2,4)
# plt.plot(x4,y4)
plt.show()


# 创建多个坐标系，调用时引用下标
fig, ax = plt.subplots(2,2)

x1 = np.linspace(-10,10,31)
y1 = x1 ** 2
ax[0,0].plot(x1,y1)

x4 = np.linspace(0,10,100)
y4 = np.cos(x4)
ax[0,1].plot(x4,y4)

x3 = np.linspace(0,10,100)
y3 = np.sin(x3)
ax[1,0].plot(x3,y3)

x2 = np.linspace(1,10,15)
y2 = 2 * x2
ax[1,1].plot(x2,y2)
plt.show()